/*
 *	Copyright (c) 2009, libBasement
 *	All rights reserved.
 *	
 *	Redistribution and use in source and binary forms, with or without modification,
 *	are permitted provided that the following conditions are met:
 *	
 *	Redistributions of source code must retain the above copyright notice,
 *	this list of conditions and the following disclaimer.
 *	Redistributions in binary form must reproduce the above copyright notice,
 *	this list of conditions and the following disclaimer in the documentation
 *	and/or other materials provided with the distribution.
 *	Neither the name of the <ORGANIZATION> nor the names of its contributors may
 *	be used to endorse or promote products derived from this software without
 *	specific prior written permission.
 *	THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 *	AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
 *	THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 *	ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE
 *	FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 *	(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 *	LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 *	ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 *	(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 *	SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include "BAutoreleasePool.h"
#include "BException.h"
#include <vector>

namespace Basement
{
	namespace AutoreleasePoolEntries
	{
		struct AutoreleaseStackEntry
		{
			AutoreleasePoolRef currentPool;
			AutoreleaseStackEntry *previousPoolEntry;
		};
		static AutoreleaseStackEntry *CurrentPool = 0;
	}
	
	typedef std::vector<Object *> ObjectVector;
	
	void AutoreleasePoolAddObject(Object *object, AutoreleasePoolRef pool)
	{
		if(object and pool)
		{
			((ObjectVector *)pool)->push_back(object);
		}
	}
	
	AutoreleasePoolRef AutoreleasePoolCreate()
	{
		AutoreleasePoolRef pool = new ObjectVector;
		
		AutoreleasePoolEntries::AutoreleaseStackEntry *entry = new AutoreleasePoolEntries::AutoreleaseStackEntry;
		entry->currentPool = pool;
		entry->previousPoolEntry = AutoreleasePoolEntries::CurrentPool;
		AutoreleasePoolEntries::CurrentPool = entry;
		
		return pool;
	}
	
	AutoreleasePoolRef AutoreleasePoolGetCurrent()
	{
		if(AutoreleasePoolEntries::CurrentPool)
		{
			return AutoreleasePoolEntries::CurrentPool->currentPool;
		}
		else
		{
			String *warning = $s("No autorelease pool in place, leaking.");
			Log::warn(warning);
			warning->release();
		}
		return NULL;
	}
	
	void AutoreleasePoolDrain(AutoreleasePoolRef pool)
	{
		for (ObjectVector::iterator iterator = ((ObjectVector *)pool)->begin(); iterator != ((ObjectVector *)pool)->end(); iterator++)
		{
			Object *object = *iterator;
			object->release();
		}
		
		delete (ObjectVector *)pool;
		
		AutoreleasePoolEntries::AutoreleaseStackEntry *currentEntry = AutoreleasePoolEntries::CurrentPool;
		AutoreleasePoolEntries::CurrentPool = currentEntry->previousPoolEntry;
		delete currentEntry;
	}
};
